#!/bin/bash
. /usr/share/openclash/openclash_ps.sh
. /lib/functions.sh
. /usr/share/openclash/ruby.sh
. /usr/share/openclash/log.sh
. /usr/share/openclash/openclash_curl.sh

   set_lock() {
      exec 877>"/tmp/lock/openclash_rule.lock" 2>/dev/null
      flock -x 877 2>/dev/null
   }

   del_lock() {
      flock -u 877 2>/dev/null
      rm -rf "/tmp/lock/openclash_rule.lock" 2>/dev/null
   }

   yml_other_rules_dl()
   {
      local section="$1"
      local enabled config
      config_get_bool "enabled" "$section" "enabled" "1"
      config_get "config" "$section" "config" ""
      
      if [ "$enabled" = "0" ] || [ "$config" != "$2" ]; then
         return
      fi
      
      if [ -n "$rule_name" ]; then
         LOG_OUT "Warrning: Multiple Other-Rules-Configurations Enabled, Ignore..."
         return
      fi
      
      config_get "rule_name" "$section" "rule_name" ""
      
      LOG_OUT "Start Downloading Third Party Rules in Use..."
      if [ "$rule_name" = "lhie1" ]; then
         if [ "$github_address_mod" != "0" ]; then
            if [ "$github_address_mod" == "https://cdn.jsdelivr.net/" ] || [ "$github_address_mod" == "https://fastly.jsdelivr.net/" ] || [ "$github_address_mod" == "https://testingcf.jsdelivr.net/" ]; then
               DOWNLOAD_URL="${github_address_mod}gh/dler-io/Rules@master/Clash/Rule.yaml"
            else
               DOWNLOAD_URL="${github_address_mod}https://raw.githubusercontent.com/dler-io/Rules/master/Clash/Rule.yaml"
            fi
         else
            DOWNLOAD_URL="https://raw.githubusercontent.com/dler-io/Rules/master/Clash/Rule.yaml"
         fi
      fi
      DOWNLOAD_FILE_CURL "$DOWNLOAD_URL" "/tmp/rules.yaml"
      if [ "$?" -eq 0 ] && [ -s "/tmp/rules.yaml" ]; then
         LOG_OUT "Download Successful, Start Preprocessing Rule File..."
         sed -i '1i rules:' /tmp/rules.yaml
         ruby -ryaml -rYAML -I "/usr/share/openclash" -E UTF-8 -e "
         begin
         YAML.load_file('/tmp/rules.yaml');
         rescue Exception => e
            YAML.LOG('Error: Unable To Parse Updated Rules File,【${rule_name}:' + e.message + '】');
            system 'rm -rf /tmp/rules.yaml 2>/dev/null';
         end
         " 2>/dev/null >> $LOG_FILE
         if [ $? -ne 0 ]; then
            LOG_OUT "Error: Ruby Works Abnormally, Please Check The Ruby Library Depends!"
            rm -rf /tmp/rules.yaml >/dev/null 2>&1
            SLOG_CLEAN
            del_lock
            exit 0
         elif [ ! -f "/tmp/rules.yaml" ]; then
            LOG_OUT "Error:【$rule_name】Rule File Format Validation Failed, Please Try Again Later..."
            rm -rf /tmp/rules.yaml >/dev/null 2>&1
            SLOG_CLEAN
            del_lock
            exit 0
         elif ! "$(ruby_read "/tmp/rules.yaml" ".key?('rules')")" ; then
            LOG_OUT "Error: Updated Others Rules【$rule_name】Has No Rules Field, Update Exit..."
            rm -rf /tmp/rules.yaml >/dev/null 2>&1
            SLOG_CLEAN
            del_lock
            exit 0
         #校验是否含有新策略组
         elif ! "$(ruby -ryaml -rYAML -I "/usr/share/openclash" -E UTF-8 -e "
            Value = YAML.load_file('/usr/share/openclash/res/${rule_name}.yaml');
            Value_1 = YAML.load_file('/tmp/rules.yaml');
            OLD_GROUP = Value['rules'].collect{|x| x.split(',')[2] or x.split(',')[1]}.uniq.map(&:strip);
            NEW_GROUP = Value_1['rules'].collect{|x| x.split(',')[2] or x.split(',')[1]}.uniq.map(&:strip);
            if (OLD_GROUP | NEW_GROUP).eql?(OLD_GROUP) then
               if (OLD_GROUP | NEW_GROUP).eql?(NEW_GROUP) then
                  puts true
               else
                  puts false
               end
            else
               puts false
            end
            ")" && [ -f "/usr/share/openclash/res/${rule_name}.yaml" ]; then
            LOG_OUT "Error: Updated Others Rules【$rule_name】Has Incompatible Proxy-Group, Update Exit, Please Wait For OpenClash Update To Adapt..."
            rm -rf /tmp/rules.yaml >/dev/null 2>&1
            SLOG_CLEAN
            del_lock
            exit 0
         fi
         
         #取出规则部分
         ruby_read "/tmp/rules.yaml" ".select {|x| 'rule-providers' == x or 'rules' == x }.to_yaml" > "$OTHER_RULE_FILE"
         #合并
         cat "$OTHER_RULE_FILE" > "/tmp/rules.yaml" 2>/dev/null
         rm -rf /tmp/other_rule* 2>/dev/null
         
         LOG_OUT "Check The Downloaded Rule File For Updates..."
         cmp -s /usr/share/openclash/res/"$rule_name".yaml /tmp/rules.yaml
         if [ "$?" -ne "0" ]; then
            LOG_OUT "Detected that The Downloaded Rule File Has Been Updated, Starting To Replace..."
            mv /tmp/rules.yaml /usr/share/openclash/res/"$rule_name".yaml >/dev/null 2>&1
            LOG_OUT "Other Rules【$rule_name】Update Successful!"
            restart=1
         else
            LOG_OUT "Updated Other Rules【$rule_name】No Change, Do Nothing!"
         fi
      else
         LOG_OUT "Other Rules【$rule_name】Update Error, Please Try Again Later..."
      fi
   }

   set_lock
   
   LOG_FILE="/tmp/openclash.log"
   RUlE_SOURCE=$(uci get openclash.config.rule_source 2>/dev/null)
   github_address_mod=$(uci -q get openclash.config.github_address_mod || echo 0)
   
   if [ "$RUlE_SOURCE" = "0" ]; then
      LOG_OUT "Other Rules Not Enable, Update Stop!"
   else
      OTHER_RULE_FILE="/tmp/other_rule.yaml"
      CONFIG_FILE=$(uci get openclash.config.config_path 2>/dev/null)
      CONFIG_NAME=$(echo "$CONFIG_FILE" |awk -F '/' '{print $5}' 2>/dev/null)
      restart=0
   
      if [ -z "$CONFIG_FILE" ]; then
         for file_name in /etc/openclash/config/*
         do
            if [ -f "$file_name" ]; then
               CONFIG_FILE=$file_name
               CONFIG_NAME=$(echo "$CONFIG_FILE" |awk -F '/' '{print $5}' 2>/dev/null)
               break
            fi
         done
      fi

      if [ -z "$CONFIG_NAME" ]; then
         CONFIG_FILE="/etc/openclash/config/config.yaml"
         CONFIG_NAME="config.yaml"
      fi
      
      config_load "openclash"
      config_foreach yml_other_rules_dl "other_rules" "$CONFIG_NAME"
      if [ -z "$rule_name" ]; then
        LOG_OUT "Get Other Rules Settings Faild, Update Stop!"
      fi
      if [ "$restart" -eq 1 ] && [ "$(unify_ps_prevent)" -eq 0 ]; then
         /etc/init.d/openclash restart >/dev/null 2>&1 &
      elif [ "$restart" -eq 0 ] && [ "$(unify_ps_prevent)" -eq 0 ] && [ "$(uci -q get openclash.config.restart)" -eq 1 ]; then
         /etc/init.d/openclash restart >/dev/null 2>&1 &
         uci -q set openclash.config.restart=0
         uci -q commit openclash
      elif [ "$restart" -eq 1 ] && [ "$(unify_ps_prevent)" -eq 0 ]; then
         uci -q set openclash.config.restart=1
         uci -q commit openclash
      fi
   fi
   rm -rf /tmp/rules.yaml >/dev/null 2>&1
   SLOG_CLEAN
   del_lock
